# 数据分析题目解答(建议先赞后看，养成习惯 如果不赞，先拉出去枪毙两分钟 作者：小匠IT)
import pandas as pd
import matplotlib.pyplot as plt
from matplotlib import font_manager as fm
import os

# 设置字体路径以支持中文显示
font_path = 'fonts/SIMSUN.TTC'
prop = fm.FontProperties(fname=font_path)

# 构造数据
data = {
    '竞品': [
        '肌肉小王子速食鸡胸肉健身代餐即食低脂卡零食鸡脯肉食品',
        '圣农去皮鸡胸肉新鲜冷冻鸡脯肉批发健身代餐单冻大胸4袋4000g8斤',
        '味小熊【酱汁100g】健身鸡胸肉即食低脂无油高蛋白营养代餐单片装',
        '小厨大乐正新口味大鸡排半成品冷冻包邮油炸小吃炸鸡排鸡扒10片',
        '小牛凯西澳洲原肉整切牛排套餐黑椒10片儿童牛肉新鲜菲力西冷20'
    ],
    '搜索人气': [2340, 9845, 5196, 8999, 12658],
    '交易指数': [10600, 37846, 50113, 73706, 93711]
}

df = pd.DataFrame(data)

# 创建输出文件夹
output_dir = 'output/15'
os.makedirs(output_dir, exist_ok=True)

# 创建散点图
plt.figure(figsize=(10, 6))
plt.scatter(df['搜索人气'], df['交易指数'])

# 添加数据标签（仅显示各竞品名称）
for i in range(len(df)):
    plt.text(df['搜索人气'][i], df['交易指数'][i], df['竞品'][i], 
             ha='center', va='bottom', fontproperties=prop, fontsize=8)

# 添加坐标轴标题
plt.xlabel('搜索人气', fontproperties=prop)
plt.ylabel('交易指数', fontproperties=prop)

# 添加图表标题
plt.title('原牧纯品竞品2021年2月搜索人气及交易指数分析图', fontproperties=prop)

# 调整布局以防止重叠
plt.tight_layout()

# 保存图表到指定文件夹
chart_output_path = os.path.join(output_dir, '原牧纯品竞品2021年2月搜索人气及交易指数分析图.png')
plt.savefig(chart_output_path, dpi=300, bbox_inches='tight')  # 保存图表为图片文件

# 显示图表
plt.show()

print(f"图表已保存至: {chart_output_path}")